import numpy as np
def 根据离散型随机变量的分布列生成一些测试数据():
    A = np.random.choice([[1, 2, 3, 4, 4]], size=100, replace=True, p=[0.1, 0.1, 0.3, 0.3, 0.2])
    print(A)
def 返回最大值最小值的索引():
    A= np.array([1,2,3,4,5])
    print(np.argmax(A),np.argmin(A))
    B = np.array([[1,2,3,4,5],[4,5,65,7,8]])
    print(np.argmax(B), np.argmin(B))
    print("对于一维矩阵返回索引，二维矩阵则是先行后列。")

if __name__=="__main__":
    返回最大值最小值的索引()